'''
SweepLEDs with a list for beeps
Option 1: count_guests_index
   plays a beep using n_guests as the index
Option 2: count_guests_random
   plays a random beep for each guest
'''
from botcore import *
from time import sleep
from random import randrange

delay = 0.1
n_led = 0
n_guests = 0
F4 = 349
C5 = 523

# List of beep frequencies
beeps = [600, 800, 1000, 1200, 1400]

def sweep_leds(n_led):
    leds.user_num(n_led, True)
    sleep(delay)
    leds.user_num(n_led, False)

# - Access beeps using its index when guest is counted
def count_guests_index(n_guests):
    freq = beeps[n_guests]
    spkr.pitch(freq)
    sleep(0.1)
    spkr.off()
    buttons.was_pressed(0)
    leds.ls_num(n_guests, True)

# - Access random beep
def count_guests_random(n_guests):
    freq = random.choice(beeps)
    spkr.pitch(freq)
    sleep(0.1)
    spkr.off()
    buttons.was_pressed(0)
    leds.ls_num(n_guests, True)

def go_straight(speed, delay):
    motors.run(LEFT, speed)
    motors.run(RIGHT, speed)
    sleep(delay)

def spin(speed):
    motors.run(LEFT, speed)
    motors.run(RIGHT, -speed)
    cute_beeps()

def cute_beeps():
    while count < 22:
        count = count + 1
        f = randrange(100, 1000)
        spkr.pitch(f)
        sleep(0.1)
    spkr.off()

def note(freq, duration):
    spkr.pitch(freq)
    sleep(duration)
    spkr.off()
    sleep(0.05)

# -- Main Program --

while True:
    sweep_leds(n_led)
    n_led = n_led + 1 
    if n_led == 8:
        n_led = 0

    if buttons.was_pressed(0):
        # Choose either count_guests_index() or count_guests_random()
        count_guests_index(n_guests)
        n_guests = n_guests + 1
        if n_guests == 5:
            break

motors.enable(True)

count = 0
go_straight(60, 3.0)
spin(30)
motors.enable(False)

# -- fanfare notes
note(F4, 0.4)
sleep(0.2)
note(F4, 0.1)
note(F4, 0.1)
note(C5, 0.8)